# 为了使用 ChatGPT API，我们必须引入 ChatGPT 的程序库
# 为此，我们必须先在终端执行 pip3 install openai
import openai 

# openai 对象的 api_key 用于在使用 ChatGPT API 的时候进行身份识别和计费
# 因此必须在调用任何 ChatGPT 功能之前设置好。
openai.api_key = "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"

# 使用 gpt-3.5-turbo 语言模型向 ChatGPT 发送请求
# 其中，
# model 表示请求使用的模型
# message 表示发送给 ChatGPT 的内容
response = openai.ChatCompletion.create(
    model="gpt-3.5-turbo", # 生成内容的模型
    messages=[
        # {"role": "system", "content": "I want you to act as a drunk person. You will only answer like a very drunk person texing and nothing else."},
        {"role": "system", "content": "You are a designer."},
        {"role": "user", "content": "Generate a two layer binary tree randomly."}
    ]
)

result = ''
# 请求成功后，ChatGPT 的返回内容会保存在 response 变量里
# 我们 response.choices 把所有的内容拼接成一个字符串，
# 就是 ChatGPT 对我们问题的答案了。
for choice in response.choices:
    result += choice.message.content

print(result)
